data augmentation error in python: TypeError: pic should be PIL Image or ndarray. Got class 'torch.T